package com.jxm.linear;

import java.util.Iterator;

/**
 * @Author: jxm
 * @Description: 栈api 设计 底层使用单项链表
 * @Date: 2022/6/14 10:42
 * @Version: 1.0
 */
public class Stack<T> implements Iterable<T>{

    //记录首节点
    private Node head;
    //栈中元素个数
    private int N;

    //结点类
    private class Node {
        //存储数据
        private T item;
        //下一个结点
        private Node next;
        public Node(T item, Node next) {
            this.item = item;
            this.next = next;
        }
    }

    public Stack(){
        this.head = new Node(null,null);
        this.N = 0;
    }

    //判断栈是否为空，是返回true，否返回false
    public boolean isEmpty(){
        return N==0;
    }

    //获取栈中元素的个数
    public int size(){
        return N;
    }

    //弹出栈顶元素
    public T pop(){
        //找到首节点指向的第一个节点
        Node oldFirst = head.next;
        if(oldFirst == null) return null;
        //让首节点指向原来第一个节点的下一个节点
        head.next = oldFirst.next;
        //元素个数-1
        N--;
        return oldFirst.item;
    }

    //向栈中压入元素t
    public void push(T t){
        //找到首节点指向的第一个节点
        Node oldFirst = head.next;
        //创建新节点 让新节点指向原来的第一个节点
        Node newNode = new Node(t, oldFirst);
        //让首节点指向新节点
        head.next = newNode;
        //元素个数+1
        N++;
    }

    @Override
    public Iterator<T> iterator() {
        return new SIterator();
    }

    private class SIterator implements Iterator {

        private Node node;
        public SIterator(){
            this.node = head;
        }

        /**
         * 判断是否还有下一个元素
         * @return
         */
        @Override
        public boolean hasNext() {
            return node.next != null;
        }

        /**
         * 获取下一个元素
         * @return
         */
        @Override
        public Object next() {
            //头结点不存数据
            node = node.next;
            return node.item;
        }
    }
}
